package com.ruoyi.common.enums;

import lombok.AllArgsConstructor;
import lombok.Getter;

public interface SparkEnum {

    /**
     * 模型为[general,generalv2,generalv3,generalv3.5,4.0Ultra]
     */
    @AllArgsConstructor
    @Getter
    enum MODEL {
        general("Lite版本", "general"),
        generalv2("V2.0版本", "generalv2"),
        generalv3("Pro版本", "generalv3"),
        generalv3_5("Max版本", "generalv3.5"),
        Ultra("4.0 Ultra版本", "4.0Ultra");

        private String alias;
        private String code;
    }

    /**
     * 工具
     */
    @AllArgsConstructor
    @Getter
    enum TOOL_CHOICE {
        NONE("不调用", "none"),
        AUTO("自动判断", "auto");

        private String alias;
        private String code;
    }
    /**
     * 对话角色
     * system用于设置对话背景，user表示是用户的问题，assistant表示AI的回复
     */
    @AllArgsConstructor
    @Getter
    enum ROLE {
        SYSTEM("系统预设", "system"),
        USER("用户问题", "user"),
        ASSISTANT("AI回复", "assistant");

        private String alias;
        private String code;
    }
}
